# ---------------------------------模型架构-----------------------------------------------
# 图聚合方法
graphEmbedMethods = ["mean", "sum", "max"]
graph_embeddings_method = graphEmbedMethods[0]
# ---------------------------------超参数设置-----------------------------------------------
# 超参数设置
lr = 0.0001
epoch = 70  # default 100
laySize = 2
embSize = 128
margin = 0.7
wordsize = 8
tar_params_name = ""
tar_params_value = None
batchSize = 1
device = "cuda:0"

# ---------------------------------训练方法-----------------------------------------------
modelName = "RGCN"
